package code;
import java.util.ArrayList;
public class GetLeastNumbers_Solution {
    public ArrayList<Integer> Solution(int [] input, int k) {
        ArrayList<Integer> re = new ArrayList<>();
        if(k>input.length){
            return re;
        }
        int index = 0;
        for(int i=0;i<input.length;i++){
            index = ifInsert(re,k,input[i]);
            if(index!=-1){
                re.add(index,input[i]);
                if(re.size()>k){
                    re.remove(re.size()-1);
                }
            }
        }
        return re;
    }

    public int ifInsert(ArrayList<Integer> re, int k,int p){
        if(re.size()==0){
            return 0;
        }
        if(re.size()==k){
            for(int i =0;i<k;i++){
                if(p<=re.get(i)){
                    return i;
                }
            }
        }else {
            int i;
            for(i =0;i<re.size();i++){
                if(p<=re.get(i)){
                    return i;
                }
            }
            if(i==re.size()){
                return re.size();
            }
        }
        return -1;

    }
}
